package com.insight.modules.system.rule;

import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.insight.common.handler.IFillRuleHandler;
import com.insight.common.util.SpringContextUtils;
import com.insight.common.util.YouBianCodeUtil;
import com.insight.common.util.oConvertUtils;
import com.insight.modules.system.entity.SysCategory;
import com.insight.modules.system.mapper.SysCategoryMapper;

import java.util.List;

/**
 * @Author scott
 * @Date 2019/12/9 11:32
 * @Description: 分类字典编码生成规则
 */
public class CategoryCodeRule implements IFillRuleHandler {

    // 根节点的父ID固定值，表示顶层节点的pid为0
    public static final String ROOT_PID_VALUE = "0";

    /**
     * 执行编码规则生成逻辑
     *
     * @param params   固定参数（可能来自页面配置）
     * @param formData 动态表单数据，通常包含当前分类的pid（父ID）
     * @return 返回生成的分类编码字符串
     */
    @Override
    public Object execute(JSONObject params, JSONObject formData) {

        // 默认将父节点ID设为根节点0
        String categoryPid = ROOT_PID_VALUE;
        String categoryCode = null;

        // 优先从动态表单参数中获取pid
        if (formData != null && formData.size() > 0) {
            Object obj = formData.get("pid");
            if (oConvertUtils.isNotEmpty(obj)) categoryPid = obj.toString();
        } else {
            // 如果formData为空，则从固定参数params中尝试获取pid
            if (params != null) {
                Object obj = params.get("pid");
                if (oConvertUtils.isNotEmpty(obj)) categoryPid = obj.toString();
            }
        }

        /*
         * 根据当前父节点的情况分为三种情况生成分类编码：
         * 1. 数据库中无子节点数据，且当前是根节点（pid=0），调用YouBianCodeUtil.getNextYouBianCode(null)获取第一个编码；
         * 2. 添加子节点，但父节点无兄弟元素，调用YouBianCodeUtil.getSubYouBianCode(parentCode,null)获取子节点编码；
         * 3. 添加子节点，且父节点有兄弟元素，调用YouBianCodeUtil.getNextYouBianCode(lastCode)获取下一个编码。
         */

        // 查询数据库中当前pid下的所有子分类，按code降序排列，取最大code
        LambdaQueryWrapper<SysCategory> query = new LambdaQueryWrapper<SysCategory>()
                .eq(SysCategory::getPid, categoryPid)
                .isNotNull(SysCategory::getCode)
                .orderByDesc(SysCategory::getCode);

        // 通过Spring上下文获取SysCategory的Mapper实例，执行查询
        SysCategoryMapper baseMapper = (SysCategoryMapper) SpringContextUtils.getBean("sysCategoryMapper");
        List<SysCategory> list = baseMapper.selectList(query);

        // 判断是否有子节点
        if (list == null || list.size() == 0) {
            // 当前pid下无子节点
            if (ROOT_PID_VALUE.equals(categoryPid)) {
                // 情况1：根节点无子节点，生成第一个编码
                categoryCode = YouBianCodeUtil.getNextYouBianCode(null);
            } else {
                // 情况2：非根节点且无兄弟节点，生成子节点编码
                SysCategory parent = (SysCategory) baseMapper.selectById(categoryPid);
                categoryCode = YouBianCodeUtil.getSubYouBianCode(parent.getCode(), null);
            }
        } else {
            // 情况3：当前pid有子节点，生成下一个编码（基于最大code）
            categoryCode = YouBianCodeUtil.getNextYouBianCode(list.get(0).getCode());
        }
        // 返回生成的分类编码
        return categoryCode;
    }
}

